package com.livew.framework.generator;

import java.util.Iterator;

import org.mybatis.generator.api.IntrospectedColumn;
import org.mybatis.generator.api.dom.xml.Attribute;
import org.mybatis.generator.api.dom.xml.TextElement;
import org.mybatis.generator.api.dom.xml.XmlElement;
import org.mybatis.generator.codegen.mybatis3.MyBatis3FormattingUtilities;
import org.mybatis.generator.codegen.mybatis3.xmlmapper.elements.AbstractXmlElementGenerator;
import org.mybatis.generator.config.PropertyRegistry;
import org.mybatis.generator.internal.util.StringUtility;

public class MySelectAllElementGenerator extends AbstractXmlElementGenerator {

	public MySelectAllElementGenerator() {
		super();
	}

	@Override
	public void addElements(XmlElement parentElement) {
		XmlElement answer = new XmlElement("select"); //$NON-NLS-1$

		answer.addAttribute(new Attribute("id", introspectedTable.getSelectAllStatementId())); //$NON-NLS-1$
		answer.addAttribute(new Attribute("resultMap", //$NON-NLS-1$
				introspectedTable.getBaseResultMapId()));

		context.getCommentGenerator().addComment(answer);

		StringBuilder sb = new StringBuilder();
		sb.append("select "); //$NON-NLS-1$
		Iterator<IntrospectedColumn> iter = introspectedTable.getAllColumns().iterator();
		while (iter.hasNext()) {
			sb.append(MyBatis3FormattingUtilities.getSelectListPhrase(iter.next()));

			if (iter.hasNext()) {
				sb.append(", "); //$NON-NLS-1$
			}

			if (sb.length() > 80) {
				answer.addElement(new TextElement(sb.toString()));
				sb.setLength(0);
			}
		}

		if (sb.length() > 0) {
			answer.addElement(new TextElement(sb.toString()));
		}

		sb.setLength(0);
		sb.append("from "); //$NON-NLS-1$
		sb.append(introspectedTable.getAliasedFullyQualifiedTableNameAtRuntime());
		sb.append(" where is_del=0 ");
		answer.addElement(new TextElement(sb.toString()));
		String orderByClause = introspectedTable
				.getTableConfigurationProperty(PropertyRegistry.TABLE_SELECT_ALL_ORDER_BY_CLAUSE);
		boolean hasOrderBy = StringUtility.stringHasValue(orderByClause);
		if (hasOrderBy) {
			sb.setLength(0);
			sb.append("order by "); //$NON-NLS-1$
			sb.append(orderByClause);
			answer.addElement(new TextElement(sb.toString()));
		}

		if (context.getPlugins().sqlMapSelectAllElementGenerated(answer, introspectedTable)) {
			parentElement.addElement(answer);
		}
	}

}
